# -*- coding: utf-8 -*-
"""
@Time    : 2021/10/11 19:25
@Author  : moqilintong

主程序，负责数据准备和数据导出
"""
from core import parseHtml
import pandas as pd

if __name__ == '__main__':
    output = []
    # URL 中 t=2 表示小学； t=3 表示中学
    url_list = {'五华区': 'https://www.ruyile.com/xuexiao/?a=4479&t=2',
                '呈贡区': 'https://www.ruyile.com/xuexiao/?a=4480&t=2',
                '官渡区': 'https://www.ruyile.com/xuexiao/?a=4482&t=2',
                '盘龙区': 'https://www.ruyile.com/xuexiao/?a=4488&t=2',
                '西山区': 'https://www.ruyile.com/xuexiao/?a=4491&t=2'}
    for name, url in url_list.items():
        # 解析网页，把数据放在 primaryschool 列表中
        parseHtml.parse_html(url, output)
        # 导出 primaryschool 中数据到指定 Excel 文档
        df = pd.DataFrame(output)
        df.to_excel('juniorschool/' + name + '.xlsx')
        output.clear()

